package com.cn.test.record;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;

/**
 * https://leetcode-cn.com/tag/tree/problemset/
 * description: 操作树节点
 *
 * @author twj
 * @date 2021/8/27 14:40 .
 */
public class OperateTreeNode {

    public static void main(String[] args) {
        TreeNode node1 = new TreeNode(1, new TreeNode(2), new TreeNode(3));
//        testinorderTraversal(node1);
        testgenerateTrees(3);
    }

    private static void testgenerateTrees(int n) {
        System.out.println(n);
        List<TreeNode> arr = generateTrees(n);
        System.out.println(arr.toString());
    }
    private static void testinorderTraversal(TreeNode node1) {
        System.out.println(node1);
        List<Integer> arr = inorderTraversal(node1);
        System.out.println(Arrays.toString(arr.toArray()));
    }


    /**
     * https://leetcode-cn.com/problems/binary-tree-inorder-traversal/
     * 给定一个二叉树的根节点 root ，返回它的 中序 遍历。
     * 输入：root = [1,null,2,3]
     * 输出：[1,3,2]
     * <p>
     * 输入：root = []
     * 输出：[]
     * <p>
     * 输入：root = [1,2]
     * 输出：[2,1]
     * <p>
     * 输入：root = [1,null,2]
     * 输出：[1,2]
     *
     * @createby twj on 2021/8/27 15:16
     */
    static List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        inorder(root, res);
        return res;
    }

    static void inorder(TreeNode root, List<Integer> res) {
        if (root == null) {
            return;
        }
        inorder(root.left, res);
        res.add(root.val);
        inorder(root.right, res);
    }

    /**
     * https://leetcode-cn.com/problems/unique-binary-search-trees-ii/
     * 给你一个整数 n ，请你生成并返回所有由 n 个节点组成且节点值从 1 到 n 互不相同的不同 二叉搜索树 。
     * 可以按 任意顺序 返回答案。
     *
     * @createby twj on 2021/8/27 16:13
     */
    static List<TreeNode> generateTrees(int n) {
        if (n == 0) {
            return new LinkedList<>();
        }
        return generateTrees(1, n);
    }

    static List<TreeNode> generateTrees(int start, int end) {
        List<TreeNode> allTrees = new LinkedList<>();
        if (start > end) {
            allTrees.add(null);
            return allTrees;
        }

        // 枚举可行根节点
        for (int i = start; i <= end; i++) {
            // 获得所有可行的左子树集合
            List<TreeNode> leftTrees = generateTrees(start, i - 1);

            // 获得所有可行的右子树集合
            List<TreeNode> rightTrees = generateTrees(i + 1, end);

            // 从左子树集合中选出一棵左子树，从右子树集合中选出一棵右子树，拼接到根节点上
            for (TreeNode left : leftTrees) {
                for (TreeNode right : rightTrees) {
                    TreeNode currTree = new TreeNode(i);
                    currTree.left = left;
                    currTree.right = right;
                    allTrees.add(currTree);
                }
            }
        }
        return allTrees;
    }

}

